Is there a function for changing all row values in a numpy array